'''
Author: SlytherinGe
LastEditTime: 2021-06-24 15:34:26
'''
import torch
import numpy as np

if __name__ == '__main__':

    data = np.ones((1,2))
    data[:,0] = 2
    data_max = np.max(data, axis=-1)
    print(data_max)